from easydict import EasyDict as ed

config = ed({
    # temp
    "data_name": "mnist",
    "arch_name": "simplenet",
    # file
    "mnist_root": "./data/mnist",
    "cifar10_root": "./data/cifar10",
    "log_path": "./log",
    "output_path": "./output",
    # learning rate
    "lr_init": 1e-2,
    "lr_max": 1e-2,
    "lr_min": 1e-5,
    # epoch
    "resume_epoch": 0,
    "warmup_epochs": 0,
    "total_epochs": 50,
    # optimizer
    "momentum": 0.9,
    "weight_decay": 1e-5,
    "nesterov": True,
})